package ai

import (
	"fmt"
	"wechat/src/service/ai/mold"
)

type Robot struct {
	Ollama *mold.Ollama `json:"ollama"`
	Expr   string       `json:"expr"`
}

func Init(expr string) *Robot {

	robot := new(Robot)
	robot.Expr = expr

	fmt.Printf("当前正在使用的是AI是：%s\n", robot.Expr)
	switch expr {
	case "ollama":
		fmt.Println("AI角色开始初始化")
		robot.Ollama = mold.OllamaInit()
		fmt.Println("AI角色初始化完毕")
	default:

	}

	return robot
}

func (robot *Robot) Send(content string, toUsername string) string {

	switch robot.Expr {
	case "ollama":
		msg, err := robot.Ollama.SendingQuestions(content, toUsername)
		if err != nil {
			return "我不知道"
		}
		return msg
	default:
		return ""
	}
}
